import pandas as pd
import os

def create_test_excel_files():
    """
    创建测试用的Excel文件
    """
    # 确保当前目录存在
    current_dir = os.path.dirname(os.path.abspath(__file__))
    
    # 创建第一个Excel文件 - 销售数据
    sales_data1 = {
        '日期': ['2023-01-01', '2023-01-02', '2023-01-03'],
        '产品': ['产品A', '产品B', '产品C'],
        '数量': [10, 15, 8],
        '金额': [1000, 2250, 800]
    }
    
    sales_data2 = {
        '日期': ['2023-01-04', '2023-01-05', '2023-01-06'],
        '产品': ['产品A', '产品B', '产品D'],
        '数量': [12, 5, 20],
        '金额': [1200, 750, 1800]
    }
    
    # 创建第二个Excel文件 - 库存数据
    inventory_data = {
        '产品编号': ['P001', 'P002', 'P003', 'P004'],
        '产品名称': ['产品A', '产品B', '产品C', '产品D'],
        '库存数量': [100, 85, 120, 50],
        '仓库位置': ['A区', 'B区', 'A区', 'C区']
    }
    
    # 创建第三个Excel文件 - 员工数据
    employee_data1 = {
        '员工ID': ['E001', 'E002', 'E003'],
        '姓名': ['张三', '李四', '王五'],
        '部门': ['销售', '市场', '技术'],
        '入职日期': ['2020-01-15', '2021-03-20', '2019-08-10']
    }
    
    employee_data2 = {
        '员工ID': ['E004', 'E005'],
        '姓名': ['赵六', '钱七'],
        '部门': ['财务', '人事'],
        '入职日期': ['2022-02-18', '2021-11-05']
    }
    
    # 保存第一个Excel文件 - 销售报表
    sales_file = os.path.join(current_dir, "销售报表.xlsx")
    with pd.ExcelWriter(sales_file) as writer:
        pd.DataFrame(sales_data1).to_excel(writer, sheet_name='一月上旬', index=False)
        pd.DataFrame(sales_data2).to_excel(writer, sheet_name='一月中旬', index=False)
    
    # 保存第二个Excel文件 - 库存报表
    inventory_file = os.path.join(current_dir, "库存报表.xlsx")
    pd.DataFrame(inventory_data).to_excel(inventory_file, index=False)
    
    # 保存第三个Excel文件 - 员工信息
    employee_file = os.path.join(current_dir, "员工信息.xlsx")
    with pd.ExcelWriter(employee_file) as writer:
        pd.DataFrame(employee_data1).to_excel(writer, sheet_name='正式员工', index=False)
        pd.DataFrame(employee_data2).to_excel(writer, sheet_name='试用期员工', index=False)
    
    print(f"已创建测试Excel文件:")
    print(f"1. {sales_file}")
    print(f"2. {inventory_file}")
    print(f"3. {employee_file}")

if __name__ == "__main__":
    create_test_excel_files() 